package com.ai.rag;

import org.springframework.ai.chat.prompt.PromptTemplate;
import org.springframework.ai.rag.generation.augmentation.ContextualQueryAugmenter;

/**
 * 创建上下文查询增强器 的工厂
 */
public class LoveAppContextualQueryAugmentFactory {

    public static ContextualQueryAugmenter createInstance(){
        PromptTemplate emptyContextPromptTemplate = new PromptTemplate(
                "你应该输出下面的内容：\n" +
                        "抱歉，我只能回答恋爱相关的问题，别的没办法帮到您哦，\n" +
                        "有问题可以联系编程导航客服"
        );
        return ContextualQueryAugmenter.builder()
                .allowEmptyContext(false) // 是否允许空上下文
                .emptyContextPromptTemplate(emptyContextPromptTemplate)
                .build();
    }
}
